Config processed at: Linux volt002 3.10.0-1160.21.1.el7.x86_64 #1 SMP Tue Mar 16 18:28:22 UTC 2021 x86_64 x86_64 x86_64 GNU/Linux
$CUDA_VISIBLE_DEVICES="0"
{ 'iotool': { 'batch_size': 4,
'collate_fn': 'CollateSparse',
'dataset': { 'data_keys': [ '../data/wire_mpvmpr_2020_04_test_small.root'],
'limit_num_files': 10,
'name': 'LArCVDataset',
'schema': { 'cluster_label': [ 'parse_cluster3d_clean_full',
'cluster3d_pcluster',
'particle_pcluster',
'particle_mpv',
'sparse3d_pcluster_semantics'],
'input_data': [ 'parse_sparse3d_scn',
'sparse3d_reco',
'sparse3d_reco_chi2'],
'kinematics_label': [ 'parse_cluster3d_kinematics_clean',
'cluster3d_pcluster',
'particle_corrected',
'particle_mpv',
'sparse3d_pcluster_semantics'],
'particle_graph': [ 'parse_particle_graph_corrected',
'particle_corrected',
'cluster3d_pcluster'],
'particles_asis': [ 'parse_particle_asis',
'particle_pcluster',
'cluster3d_pcluster'],
'particles_label': [ 'parse_particle_points_with_tagging',
'sparse3d_pcluster',
'particle_corrected'],
'segment_label': [ 'parse_sparse3d_scn',
'sparse3d_pcluster_semantics_ghost']}},
'minibatch_size': 4,
'num_workers': 4,
'shuffle': False},
'model': { 'loss_input': [ 'segment_label',
'particles_label',
'cluster_label',
'kinematics_label',
'particle_graph'],
'modules': { 'chain': { 'enable_cnn_clust': True,
'enable_cosmic': True,
'enable_ghost': True,
'enable_gnn_inter': True,
'enable_gnn_kinematics': True,
'enable_gnn_particle': False,
'enable_gnn_shower': True,
'enable_gnn_track': True,
'enable_ppn': True,
'enable_uresnet': True,
'use_ppn_in_gnn': True,
'verbose': True},
'cosmic_discriminator': { 'network_base': { 'data_dim': 3,
'features': 4,
'leakiness': 0.33,
'spatial_size': 768},
'res_encoder': { 'coordConv': True,
'latent_size': 2,
'pool_mode': 'avg'},
'uresnet_encoder': { 'features': 16,
'filters': 16,
'num_strides': 9},
'use_input_data': False,
'use_true_interactions': False},
'cosmic_loss': { 'node_loss': { 'balance_classes': True,
'name': 'type',
'target_col': 8}},
'dbscan_frag': { 'cluster_classes': [0, 2, 3],
'delta_label': 3,
'dim': 3,
'eps': [ 1.999,
3.999,
1.999,
4.999],
'michel_label': 2,
'min_samples': 1,
'min_size': [3, 3, 3, 3],
'num_classes': 4,
'ppn_distance_threshold': 1.999,
'ppn_mask_radius': 5,
'ppn_score_threshold': 0.9,
'ppn_type_threshold': 0.3,
'track_clustering_method': 'closest_path',
'track_label': 1},
'full_chain_loss': { 'clustering_weight': 1.0,
'cosmic_weight': 1.0,
'flow_weight': 1.0,
'inter_gnn_weight': 1.0,
'kinematics_p_weight': 1.0,
'kinematics_type_weight': 1.0,
'kinematics_weight': 1.0,
'particle_gnn_weight': 1.0,
'ppn_weight': 1.0,
'segmentation_weight': 1.0,
'shower_gnn_weight': 1.0,
'track_gnn_weight': 1.0},
'grappa_inter': { 'base': { 'add_start_dir': True,
'add_start_point': True,
'group_pred': 'score',
'kinematics_mlp': True,
'kinematics_momentum': False,
'kinematics_type': True,
'node_min_size': 3,
'node_type': [ 0,
1,
2,
3],
'start_dir_max_dist': 5,
'vertex_mlp': True},
'edge_encoder': { 'name': 'geo',
'use_numpy': False},
'gnn_model': { 'aggr': 'add',
'edge_classes': 2,
'edge_feats': 19,
'edge_output_feats': 64,
'leakiness': 0.1,
'name': 'meta',
'node_classes': 2,
'node_feats': 28,
'node_output_feats': 64,
'num_mp': 3},
'node_encoder': { 'name': 'geo',
'use_numpy': False},
'type_net': { 'num_hidden': 32},
'vertex_net': { 'num_hidden': 32}},
'grappa_inter_loss': { 'edge_loss': { 'balance_classes': False,
'high_purity': False,
'loss': 'CE',
'name': 'channel',
'reduction': 'sum',
'source_col': 6,
'target': 'group',
'target_col': 7},
'node_loss': { 'balance_classes': True,
'name': 'kinematics',
'spatial_size': 768,
'type_loss': 'CE'}},
'grappa_kinematics': { 'base': { 'edge_dist_metric': 'set',
'edge_dist_numpy': True,
'edge_max_dist': -1,
'kinematics_mlp': True,
'kinematics_momentum': True,
'kinematics_type': False,
'network': 'complete',
'node_min_size': -1,
'node_type': -1},
'edge_encoder': { 'cnn_encoder': { 'name': 'cnn2',
'network_base': { 'data_dim': 3,
'features': 4,
'leakiness': 0.33,
'spatial_size': 768},
'res_encoder': { 'coordConv': True,
'latent_size': 32,
'pool_mode': 'avg'},
'uresnet_encoder': { 'filters': 32,
'input_kernel': 3,
'num_classes': 5,
'num_filters': 32,
'num_strides': 9,
'reps': 2}},
'geo_encoder': { 'more_feats': True},
'name': 'mix_debug',
'normalize': True},
'gnn_model': { 'aggr': 'add',
'edge_classes': 2,
'edge_feats': 51,
'edge_output_feats': 64,
'leak': 0.33,
'name': 'nnconv_old',
'node_classes': 5,
'node_feats': 83,
'node_output_feats': 128,
'num_mp': 3},
'momentum_net': { 'num_hidden': 32},
'node_encoder': { 'cnn_encoder': { 'name': 'cnn2',
'network_base': { 'data_dim': 3,
'features': 4,
'leakiness': 0.33,
'spatial_size': 768},
'res_encoder': { 'coordConv': True,
'latent_size': 64,
'pool_mode': 'avg'},
'uresnet_encoder': { 'filters': 32,
'input_kernel': 3,
'num_classes': 5,
'num_filters': 16,
'num_strides': 9,
'reps': 2}},
'geo_encoder': { 'more_feats': True},
'name': 'mix_debug',
'normalize': True},
'use_true_particles': False},
'grappa_kinematics_loss': { 'edge_loss': { 'balance_classes': False,
'high_purity': False,
'name': 'channel',
'reduction': 'sum',
'target': 'particle_forest'},
'node_loss': { 'name': 'kinematics',
'reg_loss': 'l2'}},
'grappa_particle': { 'base': { 'node_min_size': 10,
'node_type': -1},
'edge_encoder': { 'name': 'geo',
'use_numpy': True},
'gnn_model': { 'aggr': 'add',
'edge_classes': 2,
'edge_feats': 19,
'edge_output_feats': 64,
'leakiness': 0.1,
'name': 'meta',
'node_classes': 2,
'node_feats': 24,
'node_output_feats': 64,
'num_mp': 3},
'node_encoder': { 'name': 'geo',
'use_numpy': True}},
'grappa_particle_loss': { 'edge_loss': { 'balance_classes': False,
'high_purity': True,
'loss': 'CE',
'name': 'channel',
'reduction': 'sum',
'source_col': 5,
'target': 'group',
'target_col': 6},
'node_loss': { 'balance_classes': False,
'group_pred_alg': 'score',
'high_purity': True,
'loss': 'CE',
'name': 'primary',
'reduction': 'sum',
'use_group_pred': True}},
'grappa_shower': { 'base': { 'add_start_dir': True,
'add_start_point': True,
'node_min_size': 3,
'node_type': 0,
'start_dir_max_dist': 5},
'edge_encoder': { 'name': 'geo',
'use_numpy': False},
'gnn_model': { 'aggr': 'add',
'edge_classes': 2,
'edge_feats': 19,
'edge_output_feats': 64,
'leakiness': 0.1,
'name': 'meta',
'node_classes': 2,
'node_feats': 28,
'node_output_feats': 64,
'num_mp': 3},
'node_encoder': { 'name': 'geo',
'use_numpy': False}},
'grappa_shower_loss': { 'edge_loss': { 'high_purity': True,
'name': 'channel',
'source_col': 5,
'target_col': 6},
'node_loss': { 'group_pred_alg': 'score',
'high_purity': True,
'name': 'primary',
'use_group_pred': True}},
'grappa_track': { 'base': { 'add_start_dir': True,
'add_start_point': True,
'node_min_size': 3,
'node_type': 1,
'start_dir_max_dist': 5},
'edge_encoder': { 'name': 'geo',
'use_numpy': False},
'gnn_model': { 'aggr': 'add',
'edge_classes': 2,
'edge_feats': 19,
'edge_output_feats': 64,
'leakiness': 0.1,
'name': 'meta',
'node_classes': 2,
'node_feats': 28,
'node_output_feats': 64,
'num_mp': 3},
'node_encoder': { 'name': 'geo',
'use_numpy': False}},
'grappa_track_loss': { 'edge_loss': { 'high_purity': False,
'name': 'channel',
'source_col': 5,
'target_col': 6}},
'spice': { 'fragment_clustering': { 'cluster_all': False,
'cluster_classes': [ 1],
'min_frag_size': 10,
'min_voxels': 2,
'p_thresholds': [ 0.95,
0.95,
0.95,
0.95],
's_thresholds': [ 0.0,
0.0,
0.0,
0.35]},
'network_base': { 'data_dim': 3,
'features': 4,
'leakiness': 0.33,
'spatial_size': 768},
'spatial_embeddings': { 'coordConv': True,
'embedding_dim': 3,
'seediness_dim': 1,
'sigma_dim': 1},
'uresnet': { 'filters': 64,
'input_kernel_size': 7,
'num_strides': 7,
'reps': 2}},
'spice_loss': { 'embedding_weight': 1.0,
'mask_loss_fn': 'lovasz_hinge',
'min_voxels': 2,
'name': 'se_vectorized_inter',
'seediness_weight': 1.0,
'smoothing_weight': 1.0},
'uresnet_ppn': { 'ppn': { 'classify_endpoints': True,
'data_dim': 3,
'downsample_ghost': True,
'filters': 16,
'model_name': 'ppn',
'model_path': '../data/weights_ppn3_snapshot-1999.ckpt',
'num_classes': 5,
'num_strides': 6,
'ppn1_size': 24,
'ppn2_size': 96,
'ppn_num_conv': 1,
'score_threshold': 0.5,
'spatial_size': 768,
'use_encoding': False,
'weight_ppn': 0.9},
'uresnet_lonely': { 'data_dim': 3,
'features': 2,
'filters': 16,
'freeze': False,
'ghost': True,
'leakiness': 0.0,
'num_classes': 5,
'num_strides': 6,
'spatial_size': 768,
'weight_loss': True}}},
'name': 'full_chain',
'network_input': ['input_data']},
'trainval': { 'checkpoint_step': 100,
'concat_result': [ 'seediness',
'margins',
'embeddings',
'fragments',
'fragments_seg',
'shower_fragments',
'shower_edge_index',
'shower_edge_pred',
'shower_node_pred',
'shower_group_pred',
'track_fragments',
'track_edge_index',
'track_node_pred',
'track_edge_pred',
'track_group_pred',
'particle_fragments',
'particle_edge_index',
'particle_node_pred',
'particle_edge_pred',
'particle_group_pred',
'particles',
'inter_edge_index',
'inter_node_pred',
'inter_edge_pred',
'node_pred_p',
'node_pred_type',
'flow_edge_pred',
'kinematics_particles',
'kinematics_edge_index',
'clust_fragments',
'clust_frag_seg',
'interactions',
'inter_cosmic_pred',
'node_pred_vtx',
'total_num_points',
'total_nonghost_points'],
'debug': False,
'gpus': [0],
'iterations': 652,
'log_dir': './log_trash',
'minibatch_size': -1,
'model_path': '../data/weights_full5_snapshot-999.cpkt',
'optimizer': {'args': {'lr': 0.001}, 'name': 'Adam'},
'report_step': 1,
'seed': 123,
'train': False,
'unwrapper': 'unwrap_3d_scn',
'weight_prefix': './weights_trash/snapshot'}}
Loading file: ../data/wire_mpvmpr_2020_04_test_small.root